#region Copyright 
// Copyright 2017 Gigya Inc.  All rights reserved.
// 
// Licensed under the Apache License, Version 2.0 (the "License"); 
// you may not use this file except in compliance with the License.  
// You may obtain a copy of the License at
// 
//     http://www.apache.org/licenses/LICENSE-2.0
// 
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
#endregion

using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using Gigya.Common.Contracts.HttpService;
using Gigya.Microdot.Hosting.HttpService;
using Gigya.Microdot.Interfaces;
using Orleans;

namespace Gigya.Microdot.Orleans.Hosting
{

    public class OrleansServiceInterfaceMapper : ServiceInterfaceMapper
    {
        public override IEnumerable<Type> ServiceInterfaceTypes => Mappings.Keys;

        private Dictionary<Type, Type> Mappings { get; }


        public OrleansServiceInterfaceMapper(IAssemblyProvider assemblyProvider)
        {
            Mappings = assemblyProvider.GetAllTypes()
                .Where(t => t.IsInterface && typeof(IGrain).IsAssignableFrom(t))
                .SelectMany(t => t.GetInterfaces().Where(i => i.GetCustomAttribute<HttpServiceAttribute>() != null)
                                  .Select(i => new { CallableInterface = t, ServiceInterface = i }))
                .ToDictionary(x => x.ServiceInterface, x => x.CallableInterface);


            ServiceClassesTypes = assemblyProvider.GetAllTypes().Where(t => t.IsClass && typeof(IGrain).IsAssignableFrom(t)).ToArray();

            ExtractHealthStatusServiceType(Mappings.Values);
        }


        public override Type GetGrainInterface(Type serviceInterface)
        {
            return Mappings[serviceInterface];
        }
    }
}